#pragma once

#include <srs_core.hpp>

#include <string>
#include <vector>

#ifdef SRS_AUTO_STREAM_CASTER

class SrsConfDirective;
class SrsHttpServeMux;
class SrsRtmpClient;
class SrsStSocket;
class SrsRequest;
class SrsPithyPrint;
class ISrsHttpResponseReader;
class SrsFlvDecoder;

#include <srs_app_st.hpp>
#include <srs_app_listener.hpp>
#include <srs_app_conn.hpp>
#include <srs_app_http_conn.hpp>
#include <srs_kernel_file.hpp>

/**
 * the stream caster for flv stream over HTTP POST.
 */
class SrsAppCasterFlv : virtual public ISrsTcpHandler
    , virtual public IConnectionManager, virtual public ISrsHttpHandler {
private:
    std::string output;
    SrsHttpServeMux* http_mux;
    std::vector<SrsHttpConn*> conns;
public:
    SrsAppCasterFlv(SrsConfDirective* c);
    virtual ~SrsAppCasterFlv();
public:
    virtual int initialize();
    // ISrsTcpHandler
public:
    virtual int on_tcp_client(st_netfd_t stfd);
    // IConnectionManager
public:
    virtual void remove(SrsConnection* c);
    // ISrsHttpHandler
public:
    virtual int serve_http(ISrsHttpResponseWriter* w, ISrsHttpMessage* r);
};

/**
 * the dynamic http connection, never drop the body.
 */
class SrsDynamicHttpConn : public SrsHttpConn {
private:
    std::string output;
    SrsPithyPrint* pprint;
private:
    SrsRequest* req;
    st_netfd_t stfd;
    SrsStSocket* io;
    SrsRtmpClient* client;
    int stream_id;
public:
    SrsDynamicHttpConn(IConnectionManager* cm, st_netfd_t fd, SrsHttpServeMux* m);
    virtual ~SrsDynamicHttpConn();
public:
    virtual int on_got_http_message(ISrsHttpMessage* msg);
public:
    virtual int proxy(ISrsHttpResponseWriter* w, ISrsHttpMessage* r, const std::string& o);
private:
    virtual int do_proxy(ISrsHttpResponseReader* rr, SrsFlvDecoder* dec);
    virtual int rtmp_write_packet(char type, u_int32_t timestamp, char* data, int size);
private:
    // connect to rtmp output url.
    // @remark ignore when not connected, reconnect when disconnected.
    virtual int connect();
    virtual int connect_app(const std::string& ep_server, const std::string& ep_port);
    // close the connected io and rtmp to ready to be re-connect.
    virtual void close();
};

/**
 * the http wrapper for file reader,
 * to read http post stream like a file.
 */
class SrsHttpFileReader : public SrsFileReader {
private:
    ISrsHttpResponseReader* http;
public:
    SrsHttpFileReader(ISrsHttpResponseReader* h);
    virtual ~SrsHttpFileReader();
public:
    /**
     * open file reader, can open then close then open...
     */
    virtual int open(const std::string& file);
    virtual void close();
public:
    // TODO: FIXME: extract interface.
    virtual bool is_open();
    virtual int64_t tellg();
    virtual void skip(int64_t size);
    virtual int64_t lseek(int64_t offset);
    virtual int64_t filesize();
public:
    /**
     * read from file.
     * @param pnread the output nb_read, NULL to ignore.
     */
    virtual int read(void* buf, size_t count, ssize_t* pnread);
};

#endif
